Gains and Collapse in On-Policy Distillation:A Reinforcement Learning Perspective
On-Policy 蒸馏中的收益与崩溃:强化学习视角
🧭 LLM 后训练 · On-Policy蒸馏 · RL视角
Zhejiang University · Westlake University
用 RL 视角解释 OPD 收益与崩溃:教师隐式奖励会放大学生的已有采样行为,掩码不健康响应与 SFT 初始化可缓解崩溃。
📌 工作参考价值(4 点)
- 做 LLM 后训练/蒸馏(如生成式推荐文案、Agent 轨迹优化)时,不要把教师模型只当作生成器,要同时审计它作为隐式 reward model 的可靠性;若教师打分和业务质量指标失配,可能系统性放大拼凑长文、重复槽位等低质输出。
- 在 on-policy 训练或在线蒸馏流程中加入「不健康响应 mask」:对超长、重复、包含异常模板的 rollout 直接屏蔽或降权,可低成本遏制 reward hacking。
- 上线蒸馏前先做 SFT 初始化,再切换 on-policy 蒸馏;尤其适合电商推荐解释、搜索 agent 工具调用轨迹等冷启动场景,避免从随机策略开始导致退化。
- 监控 rollout 长度分布、重复 n-gram 比例、KL 偏离等指标,作为早期预警;一旦长度/重复度飙升,优先检查教师偏好是否被错误行为劫持。